package cn.net.sinodata.generator.helper;

import cn.net.sinodata.common.core.exception.ServiceException;
import cn.net.sinodata.common.core.utils.SpringUtils;
import cn.net.sinodata.generator.enums.DataBaseType;
import lombok.AccessLevel;
import lombok.NoArgsConstructor;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

/**
 * 数据库助手
 *
 * @author yinliangyun
 */
@NoArgsConstructor(access = AccessLevel.PRIVATE)
public class DataBaseHelper {

    private static final DataSource DS = SpringUtils.getBean(DataSource.class);

    /**
     * 获取当前数据源对应的数据库类型
     * <p>
     * 获取当前线程绑定的数据源，
     * 然后从数据源获取数据库连接，利用连接的元数据获取数据库产品名称，
     * 最后调用 DataBaseType.find 方法将数据库名称转换为对应的枚举类型
     *
     * @return 当前数据库对应的 DataBaseType 枚举，找不到时默认返回 MY_SQL
     * @throws ServiceException 当获取数据库连接或元数据出现异常时抛出业务异常
     */
    public static DataBaseType getDataBaseType() {
        try (Connection conn = DS.getConnection()) {
            DatabaseMetaData metaData = conn.getMetaData();
            String databaseProductName = metaData.getDatabaseProductName();
            return DataBaseType.find(databaseProductName);
        } catch (SQLException e) {
            throw new ServiceException(e.getMessage());
        }
    }

    /**
     * 获取当前加载的数据库名，多数据源模式需要动态获取，这里先写固定值
     */
    public static List<String> getDataSourceNameList() {
        List<String> result = new ArrayList<>();
        result.add("master");
        return result;
    }

}
